#自定义一个统计目录大小的函数
import os

#函数
def dir_size(dir):
	#获取指定目录下的所有文件信息
	total = 0
	#判断指定目录是文件还是目录
	if os.path.isfile(dir):
		total+=os.path.getsize(dir)  
	if os.path.isdir(dir):
		#获取指定目录下的所有文件
		dlist = os.listdir(dir)
		#遍历目录的每一个文件
		for f in dlist:
			#为遍历的文件添加目录路径
			file = os.path.join(dir,f)
			if os.path.isfile(file):
				#获取文件大小信息并叠加
				total+=os.path.getsize(file)
			if os.path.isdir(file):
				#递归调用自己，来统计子目录文件大小
				total+=dir_size(file)
	return total

#测试
while True:
	input_address = input("请输入指定目录：")
	if input_address == "":
		break
	else:
		print("该目录大小为{}字节".format(dir_size(input_address)))


